package com.share.common.utils;

import java.util.HashMap;
import java.util.Map;

/**
 * 生成统一社会信用代码
 * <pre>
 * 第一部分：登记管理部门代码1位 (数字或大写英文字母)
 * 第二部分：机构类别代码1位 (数字或大写英文字母)
 * 第三部分：登记管理机关行政区划码6位 (数字)
 * 第四部分：主体标识码（组织机构代码）9位 (数字或大写英文字母)
 * 第五部分：校验码1位 (数字或大写英文字母)
 * </pre>
 *
 * @return 校验结果
 */
public class CreditCodeUtil {

    private static final int[] WEIGHT = {1, 3, 9, 27, 19, 26, 16, 17, 20, 29, 25, 13, 8, 24, 10, 30, 28};
    private static final char[] BASE_CODE_ARRAY = "0123456789ABCDEFGHJKLMNPQRTUWXY".toCharArray();
    private static final Map<Character, Integer> CODE_INDEX_MAP;

    static {
        CODE_INDEX_MAP = new HashMap<>();
        for (int i = 0; i < BASE_CODE_ARRAY.length; i++) {
            CODE_INDEX_MAP.put(BASE_CODE_ARRAY[i], i);
        }
    }

    /**
     * 获取一个随机的统一社会信用代码
     *
     * @return 统一社会信用代码
     */
    public static String getCreditCode() {
        final StringBuilder buf = new StringBuilder(18);

        // * 第一部分：登记管理部门代码1位 (数字或大写英文字母)
        // * 第二部分：机构类别代码1位 (数字或大写英文字母)
        for (int i = 0; i < 2; i++) {
            int num = RandomUtil.getRandom(BASE_CODE_ARRAY.length);
            buf.append(Character.toUpperCase(BASE_CODE_ARRAY[num]));
        }

        //第三部分：登记管理机关行政区划码6位 (数字)
        for (int i = 2; i < 8; i++) {
            int num = RandomUtil.getRandom(10);
            buf.append(BASE_CODE_ARRAY[num]);
        }


        // * 第四部分：主体标识码（组织机构代码）9位 (数字或大写英文字母)
        for (int i = 8; i < 17; i++) {
            int num = RandomUtil.getRandom(BASE_CODE_ARRAY.length);
            buf.append(BASE_CODE_ARRAY[num]);
        }


        String code = buf.toString();
        // * 第五部分：校验码1位 (数字或大写英文字母)
        char check = BASE_CODE_ARRAY[getParityBit(code)];
        return code + check;
    }

    /**
     * 获取校验码
     *
     * @param creditCode 统一社会信息代码
     * @return 获取较验位的值
     */
    private static int getParityBit(CharSequence creditCode) {
        int sum = 0;
        Integer codeIndex;
        for (int i = 0; i < 17; i++) {
            codeIndex = CODE_INDEX_MAP.get(creditCode.charAt(i));
            if (null == codeIndex) {
                return -1;
            }
            sum += codeIndex * WEIGHT[i];
        }
        final int result = 31 - sum % 31;
        return result == 31 ? 0 : result;
    }

    public static void main(String[] args) {
        for (int i = 0; i < 50; i++) {
            String res = getCreditCode();
            System.out.println("统一社会信用代码：" + res + "  组织机构代码:" + res.substring(8, 17));
        }
    }
}
